Week 4 graded assignment solution